#ifndef GENERAL_H
#define GENERAL_H

#include <iostream>
#include <opencv2/opencv.hpp>
#include <math.h>

using namespace std;
using namespace cv;
using namespace ml;

enum ArmorType
{
    SMALL_ARMOR = 0,
    BIG_ARMOR = 1
};

enum Color
{
    BLUE = 0,
    GREEN = 1,
    RED = 2
};

float getPointDistance(const Point2f& a, const Point2f& b);

#endif // GENERAL_H
